package com.example.controller;

import com.example.pojo.entity.JobSeeker;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;

import java.io.File;
import java.io.IOException;
import java.util.UUID;

/**
 * 所学内容:
 *
 * @author ..
 * @Date 2023/11/2 23:19
 */
@RestController
@RequestMapping("picture")
public class Message {



    @PostMapping("/getpicture/{id}")
    public String uploadImage(@RequestParam("file") MultipartFile file, @PathVariable("id") Integer id) {
        try {
            // 获取当前项目所在路径
            String rootPath = new File("").getAbsolutePath();
            System.out.println(rootPath);
//            System.out.println(10/0);
            // 设置保存图片的目录
            String uploadDir = rootPath + "/src/main/webapp/assets/img/chat/";

            // 如果目录不存在，则创建目录
            File dir = new File(uploadDir);
            if (!dir.exists()) {
                dir.mkdirs();
            }

            // 生成新的文件名
            String fileName = UUID.randomUUID().toString() + getFileExtension(file);
            // 创建新的文件
            File dest = new File(uploadDir + fileName);
            // 将上传的文件保存到新的文件中
            file.transferTo(dest);

            String newFile = "/dev/assets/img/chat/"+fileName;
            JobSeeker jobSeeker = new JobSeeker();
            jobSeeker.setId(id);
            jobSeeker.setAvatar(newFile);

            // 返回保存文件的路径
            return newFile;
        } catch (IOException e) {
            e.printStackTrace();
            return "上传失败";
        }
    }

    private String getFileExtension(MultipartFile file) {
        String originalFileName = file.getOriginalFilename();
        return originalFileName.substring(originalFileName.lastIndexOf('.'));
    }
}
